#导入numpy模块
import numpy as np
'''创建一维的数组进行分割'''
a=np.arange(1,13)
#调用split函数进行分隔
print('传递整数  平均分隔成4份')
r=np.split(a,4,axis=0) # [array([1, 2, 3]), array([4, 5, 6]), array([7, 8, 9]), array([10, 11, 12])]
print(r)

print('传递数组  按位置分隔,1-4一份 5-6一份 7-12一份')
r=np.split(a,[4,6])
print(r)  # [array([1, 2, 3, 4]), array([5, 6]), array([ 7,  8,  9, 10, 11, 12])]

'''二维数组进行分隔'''
a=np.array([[1,2,3,4],[5,6,7,8],[9,10,11,12],[13,14,15,16]])
print(a)
print('axis=0 垂直方向 平均分隔')
r,w=np.split(a,2,axis=0) # 4行4列的平均分割成2个2行4列的
print(r)
print(w)

print('axis=0 垂直方向 按位置分隔')
r,w,k=np.split(a,[2,3],axis=0) # 1-2行一组 3行一组 4行一组
print(r)
print(w)
print(k)

print('axis=1 水平方向 平均分隔')
r,w=np.split(a,2,axis=1)  # 按列平均分成两份
print(r)
print(w)

print('axis=1 水平方向 按位置分隔')
r,w=np.split(a,[3],axis=1)
print(r)
print(w)

#使用hsplit() 水平方向分隔
print('hsplit  平均分隔')
r,w=np.hsplit(a,2)
print(r)
print(w)

print('hsplit  按位置分隔')
r,w=np.hsplit(a,[3])
print(r)
print(w)
#vsplit()
print('vsplit 平均分隔')
r,w=np.vsplit(a,2)
print(r)
print(w)

print('vsplit 位置分隔')
r,w=np.vsplit(a,[1])
print(r)
print(w)

